from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage, AIMessage

# model = ChatOpenAI(model='Qwen/Qwen2.5-7B-Instruct')
model = ChatOpenAI(model='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B')

response = model.invoke([HumanMessage(content="Hi! 我是小明，你是谁")])
print(response.content)
# 没有memory ，llm记不住
response = model.invoke([HumanMessage(content="我是谁?")])
print(response.content)

# 手动维护上下文
# response = model.invoke([
#     HumanMessage(content="Hi! 我是小明，你是谁"),
#     AIMessage(content="你好！小明同学，我是Qwen，由阿里云开发的人工智能模型。很高兴认识你！有什么问题或者需要帮助的吗？"),
#     HumanMessage(content="我是谁?")
# ])
# print(response.content)
